from typing import Optional

from pydantic import BaseModel


class ChatPayload(BaseModel):
    """
    Chat 请求Payload
    """
    # session_id
    session_id: Optional[str] = None
    # 上一条message_id
    parent_message_id: Optional[int] = None
    # 用户input query
    prompt: str

    # 是否流式响应
    stream_response: Optional[bool] = True
